def cola_supplement(stock, *args):
    """
    :param stock: dict，初始库存
    :param args: 每位客人可能喝的可乐品牌集合
    :return: dict，需要补充的可乐数量
    """
    brand_count = {}
    for person in args:
        for brand in person:
            if brand in brand_count:
                brand_count[brand] += 1
            else:
                brand_count[brand] = 1


    supplement = {}
    for brand, count in brand_count.items():
        if stock.get(brand, 0) < count:
            supplement[brand] = count - stock.get(brand, 0)

    return supplement


if __name__ == '__main__':
    # 初始库存
    initial_stock = {"可口可乐": 10, "百事可乐": 2, "零度可乐": 0, "芬达": 6, "健怡可乐": 4}

    # 每位客人可能喝的可乐品牌
    p1 = {"可口可乐", "零度可乐", "芬达", "健怡可乐"}
    p2 = {"百事可乐", "零度可乐", "健怡可乐"}
    p3 = {"可口可乐", "百事可乐", "芬达"}
    p4 = {"零度可乐", "芬达", "健怡可乐"}
    p5 = {"可口可乐", "百事可乐", "健怡可乐"}


    result = cola_supplement(initial_stock, p1, p2, p3, p4, p5)
    print("需要补充的可乐有：", result)